// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_APPCACHE_APPCACHE_WORKING_SET_H_
#define CONTENT_BROWSER_APPCACHE_APPCACHE_WORKING_SET_H_

#include <stdint.h>

#include <map>

#include "base/containers/hash_tables.h"
#include "content/common/content_export.h"
#include "url/gurl.h"

namespace content {

class AppCache;
class AppCacheGroup;
class AppCacheResponseInfo;

// Represents the working set of appcache object instances
// currently in memory.
class CONTENT_EXPORT AppCacheWorkingSet {
public:
    typedef std::map<GURL, AppCacheGroup*> GroupMap;

    AppCacheWorkingSet();
    ~AppCacheWorkingSet();

    void Disable();
    bool is_disabled() const { return is_disabled_; }

    void AddCache(AppCache* cache);
    void RemoveCache(AppCache* cache);
    AppCache* GetCache(int64_t id)
    {
        CacheMap::iterator it = caches_.find(id);
        return (it != caches_.end()) ? it->second : NULL;
    }

    void AddGroup(AppCacheGroup* group);
    void RemoveGroup(AppCacheGroup* group);
    AppCacheGroup* GetGroup(const GURL& manifest_url)
    {
        GroupMap::iterator it = groups_.find(manifest_url);
        return (it != groups_.end()) ? it->second : NULL;
    }

    const GroupMap* GetGroupsInOrigin(const GURL& origin_url)
    {
        return GetMutableGroupsInOrigin(origin_url);
    }

    void AddResponseInfo(AppCacheResponseInfo* response_info);
    void RemoveResponseInfo(AppCacheResponseInfo* response_info);
    AppCacheResponseInfo* GetResponseInfo(int64_t id)
    {
        ResponseInfoMap::iterator it = response_infos_.find(id);
        return (it != response_infos_.end()) ? it->second : NULL;
    }

private:
    typedef base::hash_map<int64_t, AppCache*> CacheMap;
    typedef std::map<GURL, GroupMap> GroupsByOriginMap;
    typedef base::hash_map<int64_t, AppCacheResponseInfo*> ResponseInfoMap;

    GroupMap* GetMutableGroupsInOrigin(const GURL& origin_url)
    {
        GroupsByOriginMap::iterator it = groups_by_origin_.find(origin_url);
        return (it != groups_by_origin_.end()) ? &it->second : NULL;
    }

    CacheMap caches_;
    GroupMap groups_;
    GroupsByOriginMap groups_by_origin_; // origin -> (manifest -> group)
    ResponseInfoMap response_infos_;
    bool is_disabled_;
};

} // namespace content

#endif // CONTENT_BROWSER_APPCACHE_APPCACHE_WORKING_SET_H_
